from zhipuai import ZhipuAI
import json

# 定义问题列表
question_contents = []

# 获取json文件内容
file_path = "test.json"

# 初始化一个列表来存储提取的信息
extracted_info = []


# 读取JSON文件
with open(file_path, 'r', encoding='utf-8') as file:
    data = json.load(file)

# 提取所有的question值
question_contents = [item['question'] for item in data]

# # 打印提取的问题
# for question in question_contents:
#     print(question)
print(question_contents)

# 定义writefile函数
def writefile(answer):
    # 假设您的 JSON 文件名为 data.json
    filename = 'test.json'

    # 读取 JSON 文件
    with open(filename, 'r', encoding='utf-8') as file:
        data = json.load(file)

    # 更新答案
    updated_any = False
    
    updated = False
    for item in data:
        # 找到匹配的问题，更新答案
        item['answer'] = answer
        updated = True
        updated_any = True

    # 如果有任何更新，将更新后的数据写回 JSON 文件
    if updated_any:
        with open(filename, 'w', encoding='utf-8') as file:
            json.dump(data, file, ensure_ascii=False, indent=4)

    print("答案已更新到 JSON 文件。" if updated_any else "没有问题匹配，JSON 文件未更新。")

# 调用writefile函数
writefile("测试数据")
